from typing import Optional, Literal

from pydantic import BaseModel


class GetAnalysisResponse(BaseModel):
    category: Literal["language_use", "pronunciation"]
    content: str


class PostAnalysisRequest(BaseModel):
    message_id: str
    category: Literal["language_use", "pronunciation"]
    content: str


class PostSummaryRequest(BaseModel):
    chat_id: str
    language_use_analysis: Optional[str]
    pronunciation_analysis: Optional[str]
    communication_skill_analysis: Optional[str]
